package com.test.servlet.file;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.ProgressListener;
import org.apache.commons.fileupload.disk.DiskFileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.List;
import java.util.UUID;

public class UploadFileServlet extends HttpServlet {
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        super.doGet(req, resp);
    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //
        // 文件上传注意事项
        // 1. 为了保证服务器安全，上传文件应该放在外界无法访问的目录下，比如放在WEB-INF下
        // 2. 为了防止文件覆盖现象发生，要为上传文件设置一个唯一的名字
        // 3. 要限制上传文件的大小
        // 4. 可以限制上传文件的类型，在收到上传文件名时，判断后缀名是否合法

        /**
         * 需要用到的类详解：
         * ServletFileUpload负责处理上传文件的数据，并将表单中每个输入项封装成一个FileItem对象，在使用ServletFileUpload对象解析请求是需要
         * DiskFileItemFactory对象。所以，我们需要在进行解析工作前构造好DiskFileItemFactory对象，通过ServletFileUpload对象的构造方法
         * 或者setFileItemFactory()方法设置ServletFileUpload对象的FileItemFactory属性
         */

        String method = req.getParameter("method");
        String msg = "";

        /*try {
            msg = this.detailUpload(req,resp);
        } catch (FileUploadException e) {
            e.printStackTrace();
        }*/
        try {
            msg = this.simpleUpload(req,resp);
        } catch (FileUploadException e) {
            e.printStackTrace();
        }
        if(msg == ""){
            // 不是文件表单，是普通的表单，不处理
            return ;
        }

        //servlet 请求发消息
        req.setAttribute("msg",msg);
        req.getRequestDispatcher("").forward(req,resp);
    }

    public String detailUpload(HttpServletRequest req, HttpServletResponse resp) throws IOException, FileUploadException {

        // 判断是普通的表单还是带文件的表单
        if(!ServletFileUpload.isMultipartContent(req)){
            //不是文件表单，是普通的表单
            return "";
        }
        //创建保存路径
        String uploadPath = this.getServletContext().getRealPath("/WEB-INF/upload");
        File uploadFile = new File(uploadPath);
        if(!uploadFile.exists()){
            uploadFile.mkdir(); //创建这个目录
        }
        //缓存，临时文件
        // 临时路径，假如文件超过了预期大小，我们就把它放到一个临时文件中，过几天自动删除，或者提醒用户转存为永久
        String tmpPath = this.getServletContext().getRealPath("/WEB-INF/tmp");
        File tmpFile = new File(tmpPath);
        if(!tmpFile.exists()){
            tmpFile.mkdir(); //创建这个目录
        }
        //处理上传的文件，一般需要通过流来获取，我们可以使用request.getInputStream()，原生态的文件上传流获取，十分麻烦
        //但是我们都建议使用apache的文件上传组件来实现，common-fileupload，它需要依赖于commons-io组件
        // 1. 创建DiskFileItemFactory对象，处理文件上传路径或者大小限制的：
        DiskFileItemFactory factory = getDiskFileItemFactory(tmpFile);
        // 2. 获取ServletFileUpload
        ServletFileUpload upload = getServletFileUpload(factory);
        // 3. 处理上传的文件
        String msg = UploadParseRequest(upload,req,uploadPath);

        return msg;
    }

    public String simpleUpload(HttpServletRequest req, HttpServletResponse resp) throws FileUploadException, IOException {
        // 判断是普通的表单还是带文件的表单
        if(!ServletFileUpload.isMultipartContent(req)){
            //不是文件表单，是普通的表单
            return "";
        }
        //创建保存路径
        String uploadPath = this.getServletContext().getRealPath("/WEB-INF/upload");
        System.out.println(uploadPath);
        File uploadFile = new File(uploadPath);
        if(!uploadFile.exists()){
            boolean createUploadDir = uploadFile.mkdir(); //创建这个目录
            System.out.println(createUploadDir);
        }
        //缓存，临时文件
        // 临时路径，假如文件超过了预期大小，我们就把它放到一个临时文件中，过几天自动删除，或者提醒用户转存为永久
        String tmpPath = this.getServletContext().getRealPath("/WEB-INF/tmp");
        System.out.println(tmpPath);
        File tmpFile = new File(tmpPath);
        if(!tmpFile.exists()){
            tmpFile.mkdir(); //创建这个目录
        }
        // 1. 创建DiskFileItemFactory对象，处理文件上传路径或者大小限制的：
        DiskFileItemFactory factory = new DiskFileItemFactory();
        factory.setRepository(tmpFile);
        // 2. 获取ServletFileUpload
        ServletFileUpload upload = new ServletFileUpload(factory);
        // 3. 处理上传的文件
        List<FileItem> fileItems = upload.parseRequest(req);
        for (FileItem fileItem : fileItems) {
            if(fileItem.isFormField()){
                //普通表单
                String name = fileItem.getFieldName();
                String value = fileItem.getString("UTF-8"); //处理乱码
                System.out.println(name + ":" + value);
            }else{
                // =================== 处理文件 ========================== //
                //拿到文件名字
                String uploadFileName = fileItem.getName();
                System.out.println("上传文件名：" + uploadFileName);

                if(uploadFileName.trim().equals("") || uploadFileName == null){
                    continue;
                }
                //获得上传的文件名
                String fileName = uploadFileName.substring(uploadFileName.lastIndexOf("/") + 1);
                //获得文件的后缀
                String fileExtName = uploadFileName.substring(uploadFileName.lastIndexOf(".") + 1);
                /*
                  如果文件后缀名 fileExtName 不是我们需要的
                  就直接return，不处理，告诉用户文件类型不对
                 */
                System.out.println("文件信息[文件名："+fileName+"----文件类型"+fileExtName+"]");
                //可以使用UUID(唯一识别通用码)，保证文件名唯一
                //UUID.randomUUID()，随机生成一个唯一的通用码
                String uuidPath = UUID.randomUUID().toString();
                // ================== 文件处理完毕 ======================= //
                //存到哪？uploadPath
                //文件真实存在的路径 realPath
                String realPath = uploadPath + "/" + uuidPath;
                //给每个文件创建一个对应的文件夹
                File realPathFile = new File(realPath);
                if(!realPathFile.exists()){
                    realPathFile.mkdir();
                }
                // ============= 存放地址完毕 =============================//
                //获得上传的流
                InputStream inputStream = fileItem.getInputStream();
                //创建一个文件输出流
                // realPath = 真实的文件夹
                //差了一个文件；加上输出文件的名字 + "/" + uuidFileName
                FileOutputStream fos = new FileOutputStream(realPath + "/" + fileName);
                //创建一个缓冲区
                byte[] buffer = new byte[1024 * 1024];
                //判断是否读取完毕
                int len = 0;
                //如果大于0说明还存在数据
                while((len=inputStream.read(buffer)) > 0){
                    fos.write(buffer,0,len);
                }
                //关闭流
                fos.close();
                inputStream.close();
                fileItem.delete(); //上传成功，清除临时文件
                // 文件传输完毕
            }
        }
        return "文件上传成功";
    }


    public static DiskFileItemFactory getDiskFileItemFactory(File file){
        DiskFileItemFactory factory = new DiskFileItemFactory();
        // 通过这个工厂设置一个缓冲区，当上传文件大于这个缓冲区的时候，将它放到临时文件
        factory.setSizeThreshold(1024 * 1024); //缓冲区大小为1M
        factory.setRepository(file); //临时目录的保存目录，需要一个File
        return factory;
    }

    public static ServletFileUpload getServletFileUpload(DiskFileItemFactory factory){
        ServletFileUpload upload = new ServletFileUpload();
        //监听文件上传进度
        upload.setProgressListener(new ProgressListener() {
            // pBytesRead 已经读取到的文件大小
            // pContentLength 文件大小
            public void update(long pBytesRead, long pContentLength, int pItems) {
                System.out.println("总大小：" + pContentLength + "已上传：" + pBytesRead);
            }
        });
        //处理乱码文件
        upload.setHeaderEncoding("UTF-8");
        //设置单个文件的最大值 10M
        upload.setFileSizeMax(1024 * 1024 * 10);
        //设置总共能够上传文件的大小 10M
        upload.setSizeMax(1024 * 1024 * 10);
        return upload;
    }

    public static String UploadParseRequest(ServletFileUpload upload,HttpServletRequest request,String uploadPath) throws FileUploadException, IOException {
        String msg = "";
        // 3. 把前端请求解析，封装成一个fileItem对象
        List<FileItem> fileItems = upload.parseRequest(request);
        for (FileItem fileItem : fileItems) {
            if(fileItem.isFormField()){ //判断上传的文件时普通的表单还是带文件的表单
                //getFielfName指的是前端表单控件的name;
                String name = fileItem.getFieldName();
                String value = fileItem.getString("UTF-8"); //处理乱码
                System.out.println(name + ":" + value);
            }else{ //判断它是上传的文件
                // =================== 处理文件 ========================== //
                //拿到文件名字
                String uploadFileName = fileItem.getName();
                System.out.println("上传文件名：" + uploadFileName);

                if(uploadFileName.trim().equals("") || uploadFileName == null){
                    continue;
                }
                //获得上传的文件名
                String fileName = uploadFileName.substring(uploadFileName.lastIndexOf("/") + 1);
                //获得文件的后缀
                String fileExtName = uploadFileName.substring(uploadFileName.lastIndexOf(".") + 1);
                /*
                  如果文件后缀名 fileExtName 不是我们需要的
                  就直接return，不处理，告诉用户文件类型不对
                 */
                System.out.println("文件i型奶昔[文件名："+fileName+"----文件类型"+fileExtName+"]");
                //可以使用UUID(唯一识别通用码)，保证文件名唯一
                //UUID.randomUUID()，随机生成一个唯一的通用码
                String uuidPath = UUID.randomUUID().toString();
                // ================== 文件处理完毕 ======================= //
                //存到哪？uploadPath
                //文件真实存在的路径 realPath
                String realPath = uploadPath + "/" + uuidPath;
                //给每个文件创建一个对应的文件夹
                File realPathFile = new File(realPath);
                if(!realPathFile.exists()){
                    realPathFile.mkdir();
                }
                // ============= 存放地址完毕 =============================//
                //获得上传的流
                InputStream inputStream = fileItem.getInputStream();
                //创建一个文件输出流
                // realPath = 真实的文件夹
                //差了一个文件；加上输出文件的名字 + "/" + uuidFileName
                FileOutputStream fos = new FileOutputStream(realPath + "/" + fileName);
                //创建一个缓冲区
                byte[] buffer = new byte[1024 * 1024];
                //判断是否读取完毕
                int len = 0;
                //如果大于0说明还存在数据
                while((len=inputStream.read(buffer)) > 0){
                    fos.write(buffer,0,len);
                }
                //关闭流
                fos.close();
                inputStream.close();
                msg = "文件上传成功";
                fileItem.delete(); //上传成功，清除临时文件
                // 文件传输完毕
            }
        }
        return msg;
    }
}
